package com.xyazm.common.mq;

import com.xyazm.common.constants.CoreConstant;
import com.xyazm.common.utils.MdcUtil;
import org.apache.rocketmq.common.message.Message;


/**
 * 自定义扩展的mq消息对象
 * @author xuyou
 * @date 2023年02月13日 20:48
 */
public class MQMessage extends Message {

    public MQMessage() {
        putTraceId();
    }

    private void putTraceId() {
        String traceId = MdcUtil.getTraceId();
        if (traceId != null && !"".equals(traceId)) {
            super.putUserProperty(CoreConstant.TRACE_ID, traceId);
        }
    }


}
